from .vqvae import SMPL_VQVAE, VectorQuantizer, Encoder, Decoder